package com.ruoyi.business.service.impl;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import com.ruoyi.business.domain.dto.SkTokenGroupDTO;
import com.ruoyi.common.utils.DateUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.business.mapper.SkTokenMapper;
import com.ruoyi.business.domain.SkToken;
import com.ruoyi.business.service.ISkTokenService;

/**
 * 令牌余量Service业务层处理
 * 
 * @author ruoyi
 * @date 2025-03-06
 */
@Service
public class SkTokenServiceImpl implements ISkTokenService 
{
    @Autowired
    private SkTokenMapper skTokenMapper;

    /**
     * 查询令牌余量
     * 
     * @param id 令牌余量主键
     * @return 令牌余量
     */
    @Override
    public SkToken selectSkTokenById(Long id)
    {
        return skTokenMapper.selectSkTokenById(id);
    }

    /**
     * 查询令牌余量列表
     * 
     * @param skToken 令牌余量
     * @return 令牌余量
     */
    @Override
    public List<SkToken> selectSkTokenList(SkToken skToken)
    {
        return skTokenMapper.selectSkTokenList(skToken);
    }

    /**
     * 新增令牌余量
     * 
     * @param skToken 令牌余量
     * @return 结果
     */
    @Override
    public int insertSkToken(SkToken skToken)
    {
        skToken.setCreateTime(DateUtils.getNowDate());
        return skTokenMapper.insertSkToken(skToken);
    }

    /**
     * 修改令牌余量
     * 
     * @param skToken 令牌余量
     * @return 结果
     */
    @Override
    public int updateSkToken(SkToken skToken)
    {
        skToken.setUpdateTime(DateUtils.getNowDate());
        return skTokenMapper.updateSkToken(skToken);
    }

    /**
     * 批量删除令牌余量
     * 
     * @param ids 需要删除的令牌余量主键
     * @return 结果
     */
    @Override
    public int deleteSkTokenByIds(Long[] ids)
    {
        return skTokenMapper.deleteSkTokenByIds(ids);
    }

    /**
     * 删除令牌余量信息
     * 
     * @param id 令牌余量主键
     * @return 结果
     */
    @Override
    public int deleteSkTokenById(Long id)
    {
        return skTokenMapper.deleteSkTokenById(id);
    }

    @Override
    public List<SkTokenGroupDTO> selectSkTokenGroupByTrainCode(SkToken skToken) {
        // 获取所有符合条件的令牌数据(不分页)
        List<SkToken> allTokens = skTokenMapper.selectSkTokenList(skToken);

        // 按trainCode分组
        Map<String, List<SkToken>> tokenGroups = allTokens.stream()
                .collect(Collectors.groupingBy(SkToken::getTrainCode));

        // 转换为DTO对象
        List<SkTokenGroupDTO> result = new ArrayList<>();

        for (Map.Entry<String, List<SkToken>> entry : tokenGroups.entrySet()) {
            String trainCode = entry.getKey();
            List<SkToken> tokens = entry.getValue();

            SkTokenGroupDTO groupDTO = new SkTokenGroupDTO();
            groupDTO.setTrainCode(trainCode);
            groupDTO.setTokens(tokens);

            // 取第一条记录的日期作为组的日期
            if (!tokens.isEmpty()) {
                groupDTO.setDate(tokens.get(0).getDate());
            }

            result.add(groupDTO);
        }

        return result;
    }
}
